package com.wfm.leetcode.editor.cn;

import com.wfm.common.ListNode;
import org.junit.Test;

/**
 * 【k个一组反翻转链表】
 * 总体思路是定义四个指针，pre 【start end】 next，其中start 到 end 为待翻转的k个元素
 * 首先该题head是指向第一个元素，所以需要定义一个dummy的next指向head
 * pre和end = dummy
 * start由pre得出，next由end得出
 *
 * 循环到k后断开end.next = null
 * 每次反转start到end
 * 然后把断开的连接起来start.next = next
 * 然后把pre.next和再利用翻转后的结果链接起来
 *
 *
 */
public class BM3链表中的节点每k个一组翻转25 {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        ListNode pre = dummy;
        ListNode end = dummy;
        while (end.next!=null){
            for (int i = 0; i < k&& end!=null; i++) {
                end=end.next;
            }
            if(end==null) break;
            ListNode start = pre.next;
            ListNode next = end.next;
            end.next = null;
            new BM1翻转链表24().reverseList(start);
            pre.next = end;
            start.next = next;
            pre = start;
            end = pre;
        }
        return dummy.next;
    }

    public ListNode reverseKGroup1(ListNode head, int k) {
        ListNode dummy = new ListNode();
        dummy.next = head;
        ListNode pre = dummy;
        ListNode end = dummy;
        while (end!=null){
            while (end!=null&&k-->0){
                end=end.next;
            }

//            for (int i = 0; i < k&& end!=null; i++) {
//                end=end.next;
//            }
            if(end==null) break;
            ListNode next = end.next;
            ListNode start = pre.next;
            end.next = null;
            new BM1翻转链表24().reverseList1(pre.next);
            pre.next = end;
            start.next = next;
            pre = start;
            end = pre;
        }
        return dummy.next;
    }
    @Test
    public void test25(){
        ListNode listNode1 = new ListNode(1);
        ListNode listNode2 = new ListNode(2);
        ListNode listNode3 = new ListNode(3);
        ListNode listNode4 = new ListNode(4);
        ListNode listNode5 = new ListNode(5);;
        listNode1.next=listNode2;
        listNode2.next=listNode3;
        listNode3.next=listNode4;
        listNode4.next=listNode5;
        listNode5.next=null;
        ListNode.printListNode(reverseKGroup(listNode1, 3));
    }
}
